import time

from paho.mqtt import client as mqtt_client


# broker = "nf18a2d5.ala.cn-hangzhou.emqxsl.cn"
# broker = "tdaa11fd.ala.dedicated.aliyun.emqxcloud.cn"
broker = "broker.emqx.io"
port = 1883
topic = "test"
# generate client ID with pub prefix randomly
client_id = "ztc"
username = "ztc"
password = "1234"


def connect_mqtt():
    def on_connect(client, userdata, flags, rc):
        if rc == 0:
            print("已连接")
        else:
            print("Failed to connect, return code %d\n", rc)

    client = mqtt_client.Client(mqtt_client.CallbackAPIVersion.VERSION1, client_id)
    client.username_pw_set(username, password)
    client.on_connect = on_connect
    client.connect(broker, port)
    return client


def publish(client):
    msg_count = 0
    while True:
        time.sleep(1)
        msg = f"messages: {msg_count}"
        result = client.publish(topic, msg)
        # result: [0, 1]
        status = result[0]
        if status == 0:
            print(f"Send `{msg}` to topic `{topic}`")
        else:
            print(f"Failed to send message to topic {topic}")
        msg_count += 1


def run():
    client = connect_mqtt()
    client.loop_start()
    publish(client)


if __name__ == "__main__":
    run()
